package com.cnfangmao.util;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Enumeration;
import java.util.zip.GZIPInputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipException;
import java.util.zip.ZipFile;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;

import de.innosystec.unrar.Archive;
import de.innosystec.unrar.rarfile.FileHeader;


public class ZipUtil {
    private static final int BUFF_SIZE = 1024 * 1024; // 1M Byte 
	   private ZipUtil() {
	        // empty
	    }
	 
	 
		/****
		 * 是否为excel文件
		 * @param strPath
		 * @return
		 */
		public static boolean isZipFile(String strPath)
		{
			boolean bResult=false;
			
			String strFileSuffix=FileUtil.getFileSuffix(strPath);
			if (strFileSuffix.equalsIgnoreCase("rar") || strFileSuffix.equalsIgnoreCase("zip") || strFileSuffix.equalsIgnoreCase("tar") || strFileSuffix.equalsIgnoreCase("gz"))
			{
				bResult=true;
			}
			
			return bResult;
		}
	    /**
	     * 解压文件
	     * 
	     * @param filePath
	     *            压缩文件路径
	     */
	    public static void unzip(String filePath) {
	        File source = new File(filePath);
	        if (source.exists()) {
	            ZipInputStream zis = null;
	            BufferedOutputStream bos = null;
	            try {
	                zis = new ZipInputStream(new FileInputStream(source));
	                ZipEntry entry = null;
	                while ((entry = zis.getNextEntry()) != null
	                        && !entry.isDirectory()) {
	                    File target = new File(source.getParent(), entry.getName());
	                    if (!target.getParentFile().exists()) {
	                        // 创建文件父目录
	                        target.getParentFile().mkdirs();
	                    }
	                    // 写入文件
	                    bos = new BufferedOutputStream(new FileOutputStream(target));
	                    int read = 0;
	                    byte[] buffer = new byte[1024 * 10];
	                    while ((read = zis.read(buffer, 0, buffer.length)) != -1) {
	                        bos.write(buffer, 0, read);
	                    }
	                    bos.flush();
	                }
	                zis.closeEntry();
	            } catch (IOException e) {
	                throw new RuntimeException(e);
	            } finally {
	                try {
						zis.close();
						 bos.close();
					} catch (Exception e) {
						// TODO Auto-generated catch block
						e.printStackTrace();
					}
	               
	            }
	        }
	    }
	    
	    /**
	     * 解压文件
	     * 
	     * @param filePath
	     *            压缩文件路径
	     */
	    @SuppressWarnings("finally")
		public static String getContentFromUnzip(InputStream in,String strEncode) {

	            ZipInputStream zis = null;
	            String strResult="";
	            
	            try 
	            {
	                zis = new ZipInputStream(in);
	                ZipEntry entry = null;
	                while ((entry = zis.getNextEntry()) != null
	                        && !entry.isDirectory()) {

	                    int read = 0;
	                    byte[] buffer = new byte[1024 * 10];
	                    while ((read = zis.read(buffer, 0, buffer.length)) != -1) 
	                    {
	                    	String strContent=new String(buffer,strEncode);
	                    	strResult+=strContent;
	                    }
	                    
	                }
	                zis.closeEntry();

	            } 
	            catch (IOException e) 
	            {
	                throw new RuntimeException(e);
	            } 
	            finally 
	            {
	                try {
						zis.close();
					} catch (Exception e) {
						// TODO Auto-generated catch block
						e.printStackTrace();
					}
	                return strResult;
	            }
	        }
	    
	    /** 
	     * 数据解压缩 
	     *  
	     * @param is 
	     * @param os 
	     * @throws Exception 
	     */  
	    public static String decompressGzip(InputStream is,String strEncode)  
	    {  
	        String strResult="";
	        
	    	try
	    	{
		        GZIPInputStream gis = new GZIPInputStream(is);  
		        
		        int count=0;  
		        int nStart=0;
		        byte btCnt[] = new byte[1024*1024];  
		        byte buffer[]=new byte[1024*5];
		        while ((count = gis.read(buffer, 0, 1024*5)) != -1) 
		        {  
		        	System.arraycopy(buffer, 0, btCnt, nStart, count);
		        	nStart+=count;
		        }  
		  
            	strResult=new String(btCnt,strEncode);
 		        gis.close();  
	    	}
	    	catch(Exception e)
	    	{
	    		e.printStackTrace();
	    	}
	        return strResult;
	    }  
	    

	    
	    /**
	     * 批量压缩文件（夹）
	     * 
	     * @param resFileList
	     *            要压缩的文件（夹）列表
	     * @param zipFile
	     *            生成的压缩文件
	     * @throws IOException
	     *             当压缩过程出错时抛出
	     */ 
	    public static void zipFiles(Collection<File> resFileList, File zipFile) throws IOException { 
	 
	        ZipOutputStream zipout = null; 
	        try { 
	            zipout = new ZipOutputStream(new BufferedOutputStream(new FileOutputStream(zipFile), BUFF_SIZE)); 
	            for (File resFile : resFileList) { 
	                zipFile(resFile, zipout, ""); 
	            } 
	        } finally { 
	            if (zipout != null) 
	                zipout.close(); 
	        } 
	    } 
	 
	    /**
	     * 批量压缩文件（夹）
	     * 
	     * @param resFileList
	     *            要压缩的文件（夹）列表
	     * @param zipFile
	     *            生成的压缩文件
	     * @param comment
	     *            压缩文件的注释
	     * @throws IOException
	     *             当压缩过程出错时抛出
	     */ 
	    public static void zipFiles(Collection<File> resFileList, File zipFile, String comment) throws IOException { 
	        ZipOutputStream zipout = null; 
	        try { 
	            zipout = new ZipOutputStream(new BufferedOutputStream(new FileOutputStream(zipFile), BUFF_SIZE)); 
	            for (File resFile : resFileList) { 
	                zipFile(resFile, zipout, ""); 
	            } 
	            zipout.setComment(comment); 
	 
	        } finally { 
	            if (zipout != null) 
	                zipout.close(); 
	        } 
	    } 
	 
	    /**
	     * 解压缩一个文件
	     * 
	     * @param zipFile
	     *            压缩文件
	     * @param folderPath
	     *            解压缩的目标目录
	     * @throws IOException
	     *             当解压缩过程出错时抛出
	     */ 
	    private static void upZipFile(String zipFile, String folderPath) throws ZipException, IOException { 
	        File desDir = new File(folderPath); 
	        if (!desDir.exists()) { 
	            desDir.mkdirs(); 
	        } 
	        ZipFile zf = new ZipFile(zipFile); 
	        InputStream in = null; 
	        OutputStream out = null; 
	        try 
	        { 
	            for (Enumeration<?> entries = zf.entries(); entries.hasMoreElements();) { 
	                ZipEntry entry = ((ZipEntry) entries.nextElement()); 
	                in = zf.getInputStream(entry); 
	                String str = folderPath + File.separator + entry.getName(); 
	                str = new String(str.getBytes("8859_1"), "UTF-8"); 
	                File desFile = new File(str); 
	                if (!desFile.exists()) 
	                { 
	                    File fileParentDir = desFile.getParentFile(); 
	                    if (!fileParentDir.exists()) { 
	                        fileParentDir.mkdirs(); 
	                    } 
	                    desFile.createNewFile(); 
	                } 
	                out = new FileOutputStream(desFile); 
	                byte buffer[] = new byte[BUFF_SIZE]; 
	                int realLength; 
	                while ((realLength = in.read(buffer)) > 0) { 
	                    out.write(buffer, 0, realLength); 
	                } 
	 
	            } 
	        } finally { 
	            if (in != null) 
	                in.close(); 
	            if (out != null) 
	                out.close(); 
	        } 
	    } 
	 
	    /**   
	     * 解压rar格式压缩包。   
	     * 对应的是java-unrar-0.3.jar，但是java-unrar-0.3.jar又会用到commons-logging-1.1.1.jar   
	     */    
	    private static void unrar(String sourceRar,String destDir) throws Exception{     
	        Archive a = null;     
	        FileOutputStream fos = null;    
	        
	        File desDir = new File(destDir); 
	        if (!desDir.exists()) { 
	            desDir.mkdirs(); 
	        } 
	        try{     
	            a = new Archive(new File(sourceRar));     
	            FileHeader fh = a.nextFileHeader();     
	            while(fh!=null){     
	                if(!fh.isDirectory()){     
	                    //1 根据不同的操作系统拿到相应的 destDirName 和 destFileName     
	                    String compressFileName = fh.getFileNameString().trim();     
	                    String destFileName = "";     
	                    String destDirName = "";     
	                    //非windows系统     
	                    if(File.separator.equals("/")){     
	                        destFileName = destDir + compressFileName.replaceAll("\\\\", "/");     
	                        destDirName = destFileName.substring(0, destFileName.lastIndexOf("/"));     
	                    //windows系统      
	                    }else{     
	                        destFileName = destDir + File.separatorChar+compressFileName.replaceAll("/", "\\\\");     
	                        destDirName = destFileName.substring(0, destFileName.lastIndexOf("\\"));     
	                    }     
	                    //2创建文件夹     
	                    File dir = new File(destDirName);     
	                    if(!dir.exists()||!dir.isDirectory()){     
	                        dir.mkdirs();     
	                    }     
	                    //3解压缩文件     
	                    fos = new FileOutputStream(new File(destFileName));     
	                    a.extractFile(fh, fos);     
	                    fos.close();     
	                    fos = null;     
	                }     
	                fh = a.nextFileHeader();     
	            }     
	            a.close();     
	            a = null;     
	        }catch(Exception e){     
	            throw e;     
	        }finally{     
	            if(fos!=null){     
	                try{fos.close();fos=null;}catch(Exception e){e.printStackTrace();}     
	            }     
	            if(a!=null){     
	                try{a.close();a=null;}catch(Exception e){e.printStackTrace();}     
	            }     
	        }     
	    }  
	    
	    /****
	     * 解压缩文件
	     * @param sourceRar
	     * @param destDir
	     * @return
	     */
	    public static boolean DecompressFile(String sourceFile,String destDir)
	    {
	    	boolean bResult=false;
	    	
	    	try
	    	{
		    	String strSuffix=FileUtil.getFileSuffix(sourceFile);
		    	if (strSuffix.equalsIgnoreCase("rar"))
		    	{
		    		unrar(sourceFile,destDir);
		    	}
		    	else
		    	{
		    		//upZipFile(sourceFile,destDir);
		    		AntZipUtil.unZip(sourceFile,destDir);
		    	}
		    	bResult=true;
	    	}
	    	catch(Exception e)
	    	{
	    		e.printStackTrace();
	    	}
	    	return bResult;
	    }
	    /**
	     * 解压文件名包含传入文字的文件
	     * 
	     * @param zipFile
	     *            压缩文件
	     * @param folderPath
	     *            目标文件夹
	     * @param nameContains
	     *            传入的文件匹配名
	     * @throws ZipException
	     *             压缩格式有误时抛出
	     * @throws IOException
	     *             IO错误时抛出
	     */ 
	    public static ArrayList<File> upZipSelectedFile(File zipFile, String folderPath, String nameContains) throws ZipException, 
	            IOException { 
	        ArrayList<File> fileList = new ArrayList<File>(); 
	 
	        File desDir = new File(folderPath); 
	        if (!desDir.exists()) { 
	            desDir.mkdir(); 
	        } 
	 
	        ZipFile zf = new ZipFile(zipFile); 
	        InputStream in = null; 
	        OutputStream out = null; 
	        try { 
	            for (Enumeration<?> entries = zf.entries(); entries.hasMoreElements();) { 
	                ZipEntry entry = ((ZipEntry) entries.nextElement()); 
	                if (entry.getName().contains(nameContains)) { 
	                    in = zf.getInputStream(entry); 
	                    String str = folderPath + File.separator + entry.getName(); 
	                    str = new String(str.getBytes("8859_1"), "UTF-8"); 
	                    // str.getBytes(AppConstans.UTF_8),"8859_1" 输出 
	                    // str.getBytes("8859_1"),AppConstans.UTF_8 输入 
	                    File desFile = new File(str); 
	                    if (!desFile.exists()) { 
	                        File fileParentDir = desFile.getParentFile(); 
	                        if (!fileParentDir.exists()) { 
	                            fileParentDir.mkdirs(); 
	                        } 
	                        desFile.createNewFile(); 
	                    } 
	                    out = new FileOutputStream(desFile); 
	                    byte buffer[] = new byte[BUFF_SIZE]; 
	                    int realLength; 
	                    while ((realLength = in.read(buffer)) > 0) { 
	                        out.write(buffer, 0, realLength); 
	                    } 
	 
	                    fileList.add(desFile); 
	                } 
	            } 
	        } finally { 
	            if (in != null) 
	                in.close(); 
	            if (out != null) 
	                out.close(); 
	        } 
	        return fileList; 
	    } 
	 
	    /**
	     * 获得压缩文件内文件列表
	     * 
	     * @param zipFile
	     *            压缩文件
	     * @return 压缩文件内文件名称
	     * @throws ZipException
	     *             压缩文件格式有误时抛出
	     * @throws IOException
	     *             当解压缩过程出错时抛出
	     */ 
	    public static ArrayList<String> getEntriesNames(File zipFile) throws ZipException, IOException { 
	        ArrayList<String> entryNames = new ArrayList<String>(); 
	        Enumeration<?> entries = getEntriesEnumeration(zipFile); 
	        while (entries.hasMoreElements()) { 
	            ZipEntry entry = ((ZipEntry) entries.nextElement()); 
	            entryNames.add(new String(getEntryName(entry).getBytes("UTF-8"), "8859_1")); 
	        } 
	        return entryNames; 
	    } 
	 
	    /**
	     * 获得压缩文件内压缩文件对象以取得其属性
	     * 
	     * @param zipFile
	     *            压缩文件
	     * @return 返回一个压缩文件列表
	     * @throws ZipException
	     *             压缩文件格式有误时抛出
	     * @throws IOException
	     *             IO操作有误时抛出
	     */ 
	    public static Enumeration<?> getEntriesEnumeration(File zipFile) throws ZipException, IOException { 
	        ZipFile zf = new ZipFile(zipFile); 
	        return zf.entries(); 
	 
	    } 
	 
	    /**
	     * 取得压缩文件对象的注释
	     * 
	     * @param entry
	     *            压缩文件对象
	     * @return 压缩文件对象的注释
	     * @throws UnsupportedEncodingException
	     */ 
	    public static String getEntryComment(ZipEntry entry) throws UnsupportedEncodingException { 
	        return new String(entry.getComment().getBytes("UTF-8"), "8859_1"); 
	    } 
	 
	    /**
	     * 取得压缩文件对象的名称
	     * 
	     * @param entry
	     *            压缩文件对象
	     * @return 压缩文件对象的名称
	     * @throws UnsupportedEncodingException
	     */ 
	    public static String getEntryName(ZipEntry entry) throws UnsupportedEncodingException { 
	        return new String(entry.getName().getBytes("UTF-8"), "8859_1"); 
	    } 
	 
	    /**
	     * 压缩文件
	     * 
	     * @param resFile
	     *            需要压缩的文件（夹）
	     * @param zipout
	     *            压缩的目的文件
	     * @param rootpath
	     *            压缩的文件路径
	     * @throws FileNotFoundException
	     *             找不到文件时抛出
	     * @throws IOException
	     *             当压缩过程出错时抛出
	     */ 
	    private static void zipFile(File resFile, ZipOutputStream zipout, String rootpath) throws FileNotFoundException, IOException { 
	        rootpath = rootpath + (rootpath.trim().length() == 0 ? "" : File.separator) + resFile.getName(); 
	       // rootpath = new String(rootpath.getBytes("8859_1"), HTTP.UTF_8); 
	        BufferedInputStream in = null; 
	        try { 
	            if (resFile.isDirectory()) { 
	                File[] fileList = resFile.listFiles(); 
	                for (File file : fileList) { 
	                    zipFile(file, zipout, rootpath); 
	                } 
	            } else { 
	                byte buffer[] = new byte[BUFF_SIZE]; 
	                in = new BufferedInputStream(new FileInputStream(resFile), BUFF_SIZE); 
	                zipout.putNextEntry(new ZipEntry(rootpath)); 
	                int realLength; 
	                while ((realLength = in.read(buffer)) != -1) { 
	                    zipout.write(buffer, 0, realLength); 
	                } 
	                in.close(); 
	                zipout.flush(); 
	                zipout.closeEntry(); 
	            } 
	        } finally { 
	            if (in != null) 
	                in.close(); 
	            // if (zipout != null); 
	            // zipout.close(); 
	        } 
	    } 
}
